

package com.hazelcast.jet.impl.util;

import com.hazelcast.internal.serialization.DataSerializerHook;
import com.hazelcast.internal.serialization.impl.FactoryIdHelper;
import com.hazelcast.nio.serialization.DataSerializableFactory;

import static com.hazelcast.internal.serialization.impl.FactoryIdHelper.BASIC_FUNCTIONS_DS_FACTORY;
import static com.hazelcast.internal.serialization.impl.FactoryIdHelper.BASIC_FUNCTIONS_DS_FACTORY_ID;

/**
 * Serialization hook for functions
 */
@SuppressWarnings("checkstyle:javadocvariable")
public class FunctionsSerializerHook implements DataSerializerHook {
    public static final int F_ID = FactoryIdHelper.getFactoryId(BASIC_FUNCTIONS_DS_FACTORY, BASIC_FUNCTIONS_DS_FACTORY_ID);

    public static final int FUNCTION_IDENTITY = 1;

    @Override
    public int getFactoryId() {
        return F_ID;
    }

    @Override
    public DataSerializableFactory createFactory() {
        return typeId -> {
            switch (typeId) {
            case FUNCTION_IDENTITY:
                return Util.Identity.INSTANCE;
            default:
                throw new IllegalArgumentException("Undefined type: " + typeId);
            }
        };
    }
}
